How to implement decision tree in python